#include "ModuleLabel.h"

void ModuleLabel::init(QCVModuleHandlerInterface * parent)
{
	this->parent=parent;
	win=new QWidget();
  layout=new QVBoxLayout(win);
	sb_radius=new QSpinBox(win);
	sb_radius->setPrefix("radius=");
	sb_radius->setRange(0,100);	
	layout->addWidget(sb_radius);
}
////////////////////////////////////////////////////////////////////////////////
void ModuleLabel::free()
{
	delete win;
}
////////////////////////////////////////////////////////////////////////////////
cv::Mat & ModuleLabel::process()
{
	//Labelisation des points clefs celon une distance
	if(!keyspoints.empty())
	{
		cvLabelizeKeyspoints(keyspoints,sb_radius->value());
	}
	return keyspoints;
}
////////////////////////////////////////////////////////////////////////////////
void ModuleLabel::setInput(int n,const cv::Mat & m)
{
	if(n==0)
	{	
		keyspoints=m.clone();
	}
}
////////////////////////////////////////////////////////////////////////////////
QWidget* ModuleLabel::widget()
{
	return win;
}
////////////////////////////////////////////////////////////////////////////////
QString ModuleLabel::getName()
{
	return tr("Label Keyspoints");
}
////////////////////////////////////////////////////////////////////////////////
int ModuleLabel::nbInputs()
{
	return 1;
}
////////////////////////////////////////////////////////////////////////////////
bool ModuleLabel::setParams(QString params)
{
	QStringList list_params=params.split(";");
		
	if(list_params.size()==1)
	{
		sb_radius->setValue(list_params[0].toInt());
		return true;
	}
	else
	{
		return false;
	}	
}
////////////////////////////////////////////////////////////////////////////////
QString ModuleLabel::getParams()
{
	return QString::number(sb_radius->value());
}
////////////////////////////////////////////////////////////////////////////////
QString ModuleLabel::help(int id)
{
	if(id==-2)
	{
		return tr("Permet de labelizer les points clefs en fonction d'un rayon");
	}
	else if(id==-1)
	{
		return tr("keyspoints");
	}
	else if(id==0)
	{
		return tr("keyspoints");
	}
	else
	{
		return tr("erreur");
	}
}
////////////////////////////////////////////////////////////////////////////////
Q_EXPORT_PLUGIN2(ModuleLabel_plugin,ModuleLabel);  
